#!/usr/bin/env python
# coding=utf-8
import pretty_errors
import requests, re, json, time


# 3.4 抓取猫眼电影排行
# 目标页面:https://maoyan.com/board/4

# 4.抓取首页
def get_one_page(url):
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) '
                      'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.87 Safari/537.36'
    }
    response = requests.get(url, headers=headers)
    if response.status_code == 200:
        return response.text
    return None


# 5.正则提取
def parse_one_page(html):
    pattern = re.compile(
        '<dd>.*?board-index.*?>(.*?)</i>.*?data-src="(.*?)".*?'
        'name.*?a.*?>(.*?)</a>.*?star.*?>(.*?)</p>.*?releasetime.*?>'
        '(.*?)</p>.*?integer.*?>(.*?)</i>.*?fraction.*?>(.*?)</i>.*?</dd>',
        re.S)
    items = re.findall(pattern, html)
    for item in items:
        yield {  # yield 打包字典,直接返回,函数不需要另外加return,直接返回对象
            'index': item[0],
            'image': item[1],
            'title': item[2].strip(),
            'actor': item[3].strip()[3:] if len(item[3]) > 3 else '',
            'time': item[4].strip()[5:] if len(item[4]) > 5 else '',
            'score': item[5].strip() + item[6].strip()
        }


# 6.写入文件
def write_to_file(content):
    with open('result.txt', 'a', encoding='utf-8') as f:
        print(type(json.dumps(content)))
        f.write(json.dumps(content, ensure_ascii=False) + '\n')


# 7. 整合代码
def main(offset):
    url = 'https://maoyan.com/board/4?offset=' + str(offset)  # 放入分页参数
    html = get_one_page(url)
    for item in parse_one_page(html):
        print(item)
        write_to_file(item)


if __name__ == '__main__':
    for i in range(10):
        main(offset=i * 10)
        time.sleep(1)


'''
代码参考:
https://github.com/Python3WebSpider/MaoYan
'''